import os
from litellm import completion
from dotenv import load_dotenv

messages =[
    {"role":"user","content":"1+1等于多少？"},
    {"role":"assistant","content":"1+1等于2。"},
    {"role":"user","content":"我刚才说了什么？"}
]
load_dotenv()
response = completion(model=os.getenv("MODEL"),
                      base_url=os.getenv("BASE_URL"),
                      api_key=os.getenv("API_KEY"),
                      messages=messages)
print(response.choices[0].message.content)